#include "AIManagerState.h"

#include "AIState.h"
#include "AIFollowState.h"
#include "AIPatrolState.h"
#include "MeleeState.h"
#include "RangedState.h"

#include "..\Game Objects\Enemy.h"

AIManagerState::AIManagerState()
{
	// Add AI State to vector
	vAIStates.push_back(new AIFollowState);
	vAIStates.push_back(new AIPatrolState);
	vAIStates.push_back(new MeleeState);
	vAIStates.push_back(new RangedState);
}

AIManagerState::~AIManagerState()
{
	for (unsigned int i = 0; i < vAIStates.size(); i++)
	{
		delete vAIStates[i];
	}
	vAIStates.clear();
}


// Add Enemy or Boss To AIManager
//  - WhichState = 0 for Patrol, 1 for Melee, 2 for Ranged
AIState* AIManagerState::GetAIState(int iWhichState)
{
	return vAIStates[iWhichState];
}

